import torch as pt
from python_ai.common.xcommon import *

pt.set_printoptions(edgeitems=100)

sep('3x4')
x = pt.arange(1, 3*4+1).view(3, 4)
print(x)

sep('masked_select')
mask = x % 3 == 0
print(mask)
x2 = pt.masked_select(x, mask)
print(x2)
